class Config:
    CONFIG_CLASSES = (
        'not-hotdog', 'hotdog'
    )

    CONFIG_BATCH_SIZE = 4
    CONFIG_ROOT_PATH = r'H:\notebook\LeNet\note1_extra_selfdata\hotdog'
    CONFIG_SAVE_PATH = r'./lenet.pth'
    CONFIG_TEST_IMG = r'test2.jpg'
    # CONFIG_TEST_IMG = r'test.jpg'
    CONFIG_SAVE_FIG_PATH = r'out.jpg'
    CONFIG_RESIZE_SIZE = 64
